how to input in array in python